<template>
  <Dialog :title="dialogTitle" v-model="dialogVisible">
    <el-form
      ref="formRef"
      :model="formData"
      :rules="formRules"
      label-width="100px"
      v-loading="formLoading"
    >
      <el-form-item label="任务名称" prop="name">
        <el-input v-model="formData.name" placeholder="请输入任务名称" />
      </el-form-item>
      <el-form-item label="评测类型 0 单个评测，1 对比评测" prop="type">
        <el-select v-model="formData.type" placeholder="请选择评测类型 0 单个评测，1 对比评测">
          <el-option
            v-for="dict in getIntDictOptions(DICT_TYPE.MODEL_EVALUATE_TYPE)"
            :key="dict.value"
            :label="dict.label"
            :value="dict.value"
          />
        </el-select>
      </el-form-item>
      <el-form-item label="数据集集合，逗号隔开" prop="datasetIds">
        <el-input v-model="formData.datasetIds" placeholder="请输入数据集集合，逗号隔开" />
      </el-form-item>
      <el-form-item label="模型集合，逗号隔开" prop="largeModelIds">
        <el-input v-model="formData.largeModelIds" placeholder="请输入模型集合，逗号隔开" />
      </el-form-item>
      <el-form-item label="推理状态（0 待推理，1 推理中， 2 推理完成）" prop="status">
        <el-radio-group v-model="formData.status">
          <el-radio
            v-for="dict in getIntDictOptions(DICT_TYPE.MODEL_EXEC_STATUS)"
            :key="dict.value"
            :label="dict.value"
          >
            {{ dict.label }}
          </el-radio>
        </el-radio-group>
      </el-form-item>
      <el-form-item label="指标1权重" prop="indicator1Weight">
        <el-input v-model="formData.indicator1Weight" placeholder="请输入指标1权重" />
      </el-form-item>
      <el-form-item label="指标2权重" prop="indicator2Weight">
        <el-input v-model="formData.indicator2Weight" placeholder="请输入指标2权重" />
      </el-form-item>
      <el-form-item label="指标3权重" prop="indicator3Weight">
        <el-input v-model="formData.indicator3Weight" placeholder="请输入指标3权重" />
      </el-form-item>
      <el-form-item label="指标4权重" prop="indicator4Weight">
        <el-input v-model="formData.indicator4Weight" placeholder="请输入指标4权重" />
      </el-form-item>
      <el-form-item label="指标5权重" prop="indicator5Weight">
        <el-input v-model="formData.indicator5Weight" placeholder="请输入指标5权重" />
      </el-form-item>
      <el-form-item label="指标6权重" prop="indicator6Weight">
        <el-input v-model="formData.indicator6Weight" placeholder="请输入指标6权重" />
      </el-form-item>
      <el-form-item label="指标7权重" prop="indicator7Weight">
        <el-input v-model="formData.indicator7Weight" placeholder="请输入指标7权重" />
      </el-form-item>
      <el-form-item label="指标8权重" prop="indicator8Weight">
        <el-input v-model="formData.indicator8Weight" placeholder="请输入指标8权重" />
      </el-form-item>
      <el-form-item label="指标9权重" prop="indicator9Weight">
        <el-input v-model="formData.indicator9Weight" placeholder="请输入指标9权重" />
      </el-form-item>
      <el-form-item label="指标10权重" prop="indicator10Weight">
        <el-input v-model="formData.indicator10Weight" placeholder="请输入指标10权重" />
      </el-form-item>
    </el-form>
    <template #footer>
      <el-button @click="submitForm" type="primary" :disabled="formLoading">确定</el-button>
      <el-button @click="dialogVisible = false">取消</el-button>
    </template>
  </Dialog>
</template>
<script setup lang="ts">
import { getIntDictOptions, DICT_TYPE } from '@/utils/dict'
import { EvaluationTaskApi, EvaluationTaskVO } from '@/api/multiModel/evaluationtask'

/** 评测任务 表单 */
defineOptions({ name: 'EvaluationTaskForm' })

const { t } = useI18n() // 国际化
const message = useMessage() // 消息弹窗

const dialogVisible = ref(false) // 弹窗的是否展示
const dialogTitle = ref('') // 弹窗的标题
const formLoading = ref(false) // 表单的加载中：1）修改时的数据加载；2）提交的按钮禁用
const formType = ref('') // 表单的类型：create - 新增；update - 修改
const formData = ref({
  id: undefined,
  name: undefined,
  type: undefined,
  datasetIds: undefined,
  largeModelIds: undefined,
  status: undefined,
  indicator1Weight: undefined,
  indicator2Weight: undefined,
  indicator3Weight: undefined,
  indicator4Weight: undefined,
  indicator5Weight: undefined,
  indicator6Weight: undefined,
  indicator7Weight: undefined,
  indicator8Weight: undefined,
  indicator9Weight: undefined,
  indicator10Weight: undefined,
})
const formRules = reactive({
  type: [{ required: true, message: '评测类型 0 单个评测，1 对比评测不能为空', trigger: 'change' }],
  datasetIds: [{ required: true, message: '数据集集合，逗号隔开不能为空', trigger: 'blur' }],
  largeModelIds: [{ required: true, message: '模型集合，逗号隔开不能为空', trigger: 'blur' }],
  status: [{ required: true, message: '推理状态（0 待推理，1 推理中， 2 推理完成）不能为空', trigger: 'blur' }],
})
const formRef = ref() // 表单 Ref

/** 打开弹窗 */
const open = async (type: string, id?: number) => {
  dialogVisible.value = true
  dialogTitle.value = t('action.' + type)
  formType.value = type
  resetForm()
  // 修改时，设置数据
  if (id) {
    formLoading.value = true
    try {
      formData.value = await EvaluationTaskApi.getEvaluationTask(id)
    } finally {
      formLoading.value = false
    }
  }
}
defineExpose({ open }) // 提供 open 方法，用于打开弹窗

/** 提交表单 */
const emit = defineEmits(['success']) // 定义 success 事件，用于操作成功后的回调
const submitForm = async () => {
  // 校验表单
  await formRef.value.validate()
  // 提交请求
  formLoading.value = true
  try {
    const data = formData.value as unknown as EvaluationTaskVO
    if (formType.value === 'create') {
      await EvaluationTaskApi.createEvaluationTask(data)
      message.success(t('common.createSuccess'))
    } else {
      await EvaluationTaskApi.updateEvaluationTask(data)
      message.success(t('common.updateSuccess'))
    }
    dialogVisible.value = false
    // 发送操作成功的事件
    emit('success')
  } finally {
    formLoading.value = false
  }
}

/** 重置表单 */
const resetForm = () => {
  formData.value = {
    id: undefined,
    name: undefined,
    type: undefined,
    datasetIds: undefined,
    largeModelIds: undefined,
    status: undefined,
    indicator1Weight: undefined,
    indicator2Weight: undefined,
    indicator3Weight: undefined,
    indicator4Weight: undefined,
    indicator5Weight: undefined,
    indicator6Weight: undefined,
    indicator7Weight: undefined,
    indicator8Weight: undefined,
    indicator9Weight: undefined,
    indicator10Weight: undefined,
  }
  formRef.value?.resetFields()
}
</script>